# 人口数量
# 修改前面的函数，加上第三个必不可少的形参population ，并返回一个格式为 City , Country – population xxx 的字符串，如Santiago, Chile –population 5000000 。运行test_cities.py，确认测试test_city_country() 未通过。
# 修改上述函数，将形参population 设置为可选的。再次运行test_cities.py，确认测试test_city_country() 又通过了。
# 再编写一个名为test_city_country_population() 的测试，核实可以使用类似于'santiago' 、'chile' 和'population=5000000' 这样的值来调用这个函数。再次运行test_cities.py，确认测试test_city_country_population() 通过了。

import unittest

from city_functions import get_formatted_country


class CityCountryTest(unittest.TestCase):
    def test_get_formatted_country(self):
        city_country = get_formatted_country('santiago', 'chile', 500000)
        self.assertEqual("Santiago, Chile - 500000", city_country)


if __name__ == '__main__':
    unittest.main()
